How to transform the dataframe in python